package com.wxweven.algorithm.array;

import org.junit.Test;

import java.util.LinkedList;

public class LC239滑动窗口最大值 {

    private static final int[] nums = new int[]{1, 3, -1, -3, 5, 3, 6, 7};

    /*
     * 给你一个整数数组 nums，有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。
     * 滑动窗口每次只向右移动一位。返回滑动窗口中的最大值。
     *
     * 示例 1：
     *
     * 输入：nums = [1,3,-1,-3,5,3,6,7], k = 3
     * 输出：[3,3,5,5,6,7]
     * 解释：
     * 滑动窗口的位置                最大值
     * ---------------               -----
     * [1  3  -1] -3  5  3  6  7       3
     *  1 [3  -1  -3] 5  3  6  7       3
     *  1  3 [-1  -3  5] 3  6  7       5
     *  1  3  -1 [-3  5  3] 6  7       5
     *  1  3  -1  -3 [5  3  6] 7       6
     *  1  3  -1  -3  5 [3  6  7]      7
     * 示例 2：
     *
     * 输入：nums = [1], k = 1
     * 输出：[1]
     * 示例 3：
     *
     * 输入：nums = [1,-1], k = 1
     * 输出：[1,-1]
     * 示例 4：
     *
     * 输入：nums = [9,11], k = 2
     * 输出：[11]
     * 示例 5：
     *
     * 输入：nums = [4,-2], k = 2
     * 输出：[4]
     *
     */
    public int[] maxSlidingWindow(int[] nums, int k) {
        if (nums == null || nums.length == 0) {
            return new int[0];
        }

        Window window = new Window();
        int len = nums.length;

        int[] res = new int[len - k + 1];
        int index = 0;

        for (int i = 0; i < nums.length; i++) {
            if (i < k - 1) {
                window.addLast(nums[i]);
            } else {
                window.addLast(nums[i]);
                int max = window.max();
                res[index++] = max;
                window.removeFirst(nums[i - k + 1]);
            }
        }

        return res;
    }

    private static class Window {
        LinkedList<Integer> list = new LinkedList<>();

        public void addLast(int num) {
            while (!list.isEmpty() && list.getLast() < num) {
                list.pollLast();
            }

            list.addLast(num);
        }

        public int max() {
            return list.getFirst();
        }

        public void removeFirst(int num) {
            if (list.getFirst() == num) {
                list.pollFirst();
            }
        }
    }

    @Test
    public void test() {
        int[] res = maxSlidingWindow(nums, 3);
        for (int re : res) {
            System.out.print(re + " ");
        }
    }
}